/*
    提供一些工具类
*/

#pragma once
#include <iostream>
#include <string>
#include <vector>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <atomic>
#include <fstream>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    // 时间类
    class TimeUtil
    {
    public:
        static std::string GetTime()
        {
            struct timeval timeval_;
            gettimeofday(&timeval_, nullptr);
            return std::to_string(timeval_.tv_sec);
        }

        // 获取毫秒级时间戳
        static std::string GetTimeMs()
        {
            struct timeval timeval_;
            gettimeofday(&timeval_, nullptr);
            return std::to_string(timeval_.tv_sec * 1000) + std::to_string(timeval_.tv_usec / 1000);
        }
    };

    const std::string temp_path = "./temp/"; // 存放临时文件的地址

    // 对文件路径和后缀处理工具类
    class PathUtil
    {
    public:
        //  编译文件
        // 返回源文件路径
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cc");
        }

        // 返回可执行文件路径
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        // 返回编译报错文件路径
        static std::string CompileErr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_err");
        }

        // 运行文件
        // 返回输入文件路径
        static std::string RunIn(const std::string &file_name)
        {
            return AddSuffix(file_name, ".run_in");
        }

        // 返回输出文件路径
        static std::string RunOut(const std::string &file_name)
        {
            return AddSuffix(file_name, ".run_out");
        }

        // 返回报错文件路径
        static std::string RunErr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".run_err");
        }

        // 给文件名添加路径和后缀
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string file_path = temp_path;
            file_path += file_name;
            file_path += suffix;
            return file_path;
        }
    };

    // 对文件处理工具类
    class FileUtil
    {
    public:
        // 判断file_path文件是否存在
        static bool IsExists(const std::string &file_path)
        {
            struct stat statbuf; // 记录文件状态信息
            int ret = stat(file_path.c_str(), &statbuf);
            return ret == 0;
        }

        // 生成唯一文件名，不包括路径和后缀
        static std::string UniqueFileName()
        {
            std::string timestamp = TimeUtil::GetTimeMs(); // 获取毫秒级时间戳
            static std::atomic_uint id(0); // 定义原子变量
            id++;
            return timestamp + "_" + std::to_string(id);
        }

        // 将code写入文件
        static bool WriteFile(const std::string &file_path, const std::string &code)
        {
            std::ofstream out(file_path);
            if (!out.is_open())
            {
                // 文件打开失败
                return false;
            }

            out.write(code.c_str(), code.size()); // 写入文件

            out.close(); // 关闭文件
            return true;
        }

        // 读取文件file_path到content，space控制是否读取换行符
        static bool ReadFile(const std::string &file_path, std::string& content, bool space = false)
        {
            content.clear();
            std::ifstream in(file_path);
            if (!in.is_open())
            {
                return false;
            }

            std::string line;
            while(std::getline(in, line))
            {
                line += space ? "\n" : "";
                content += line;
            }

            in.close();
            return true;
        }

        // 分割字符串
        static void StringSplit(const std::string str, std::vector<std::string>& v, const std::string& sep)
        {
            boost::split(v, str, boost::is_any_of(sep), boost::token_compress_on);
        }
    };

}